"""TFR-BERT."""

# Placeholder: load py_library
# Placeholder: load py_test

package(
    default_visibility = [
        "//tensorflow_ranking:__subpackages__",
    ],
)

licenses(["notice"])

py_library(
    name = "premade",
    srcs = ["__init__.py"],
    srcs_version = "PY3",
    deps = [
        ":tfrbert_task",
    ],
)

py_library(
    name = "tfrbert_task",
    srcs = ["tfrbert_task.py"],
    deps = [
        # py/absl/logging dep,
        # py/numpy dep,
        # py/tensorflow dep,
        # tensorflow_models/official/core:task_factory dep,
        # tensorflow_models/official/modeling/hyperparams:base_config dep,
        # tensorflow_models/official/nlp/configs:encoders dep,
        "//tensorflow_ranking/extension:task",
        "//tensorflow_ranking/python/keras:metrics",
        "//tensorflow_ranking/python/keras:model",
    ],
)

py_test(
    name = "tfrbert_task_test",
    srcs = ["tfrbert_task_test.py"],
    python_version = "PY3",
    deps = [
        ":tfrbert_task",
        # py/absl/flags dep,
        # py/absl/testing:parameterized dep,
        # Placeholder: proto upb dep
        # py/tensorflow dep,
        # tensorflow_models/official/nlp/configs:bert dep,
        # tensorflow_models/official/nlp/configs:encoders dep,
        # tensorflow_serving/apis:input_proto_py_pb2 dep,
    ],
)
